# encoding: utf-8

import torch

# 示例张量
x = torch.tensor([3, 1, 4, 1, 5])

# 第一步：获取最大值的索引（不改变原始张量）
_, idx1 = torch.max(x, 0)

# 第二步：将第一个最大值设置为负无穷（或者一个比所有其他值都小的数），然后再次找到最大值
x[idx1] = -99999  # 注意：这里我们使用float('-inf')来确保最大值被替换掉
_, idx2 = torch.max(x, 0)

# 注意：由于我们修改了x，你可能想要保留原始张量的副本
# 如果需要原始张量，在第一步之前可以先复制它：x_orig = x.clone()

# 现在idx1和idx2分别包含第一个和第二个最大值的索引
print("第一个最大值的索引:", idx1.item())
print("第二个最大值的索引:", idx2.item())
